dataset:
  name: agnews
  path: data/TextClassification/agnews

plm:
  model_name: roberta
  model_path: roberta-large
  optimize:
    freeze_para: False
    lr: 0.00003
    weight_decay: 0.01
    scheduler:
      type: 
      num_warmup_steps: 500

train:
  batch_size: 2



template: manual_template
verbalizer: automatic_verbalizer

manual_template:
  choice: 1
  file_path: scripts/TextClassification/agnews/manual_template.txt


automatic_verbalizer:
  parent_config: verbalizer
  num_cadidates: 1000
  label_word_num_per_class: 1
  num_searches: 1
  score_fct: llr
  balance: true
  num_classes: 4
  init_using_split: train

environment:
  num_gpus: 1
  cuda_visible_devices:
    - 0
  local_rank: 0
  
learning_setting: few_shot

few_shot:
  parent_config: learning_setting
  few_shot_sampling: sampling_from_train
  
sampling_from_train:
  parent_config: few_shot_sampling
  num_examples_per_label: 10
  also_sample_dev: True
  num_examples_per_label_dev: 10
  seed:
    - 123


